#ifndef THREADPOOL_H
#define THREADPOOL_H

#include <pthread.h>
#include <list>
#include <exception>
#include <cstdio>
#include "../locker/locker.h"
#include "../log/log.h"
#include "../mysql_conn/sql_connection_pool.h"

extern int m_close_log;

// 线程池类
template <typename T>
class threadpool
{
public:
    threadpool(int thread_num = 9, int max_requests = 10000);
    threadpool(connection_pool *connPool);
    ~threadpool();
    bool append(T *request);

private:
    static void *worker(void *arg);
    void run();

private:
    // 线程的数量
    int m_thread_number;

    // 线程池数组
    pthread_t *m_threads;

    // 请求队列数量
    int m_max_requests;

    // 请求队列
    std::list<T *> m_workqueue;

    // 互斥锁
    locker m_queuelocker;

    // 信号量
    sem m_queuestat;

    // 是否结束线程
    bool m_stop;

    connection_pool *m_connPool; // 数据库
};

template <typename T>
threadpool<T>::threadpool(connection_pool *connPool) : threadpool()
{
    m_connPool = connPool;
}

template <typename T>
threadpool<T>::threadpool(int thread_num, int max_requests) : m_thread_number(thread_num), m_max_requests(max_requests), m_stop(false), m_threads(NULL)
{
    if (thread_num <= 0 || max_requests <= 0)
    {
        throw std::exception();
    }
    m_threads = new pthread_t[m_thread_number];
    if (!m_threads)
    {
        throw std::exception();
    }
    // 创建thread_number个线程
    for (int i = 0; i < thread_num; i++)
    {
        LOG_INFO("create the %dth thread\n", i);
        if (pthread_create(m_threads + i, NULL, worker, this) != 0)
        {
            delete[] m_threads;
            throw std::exception();
        }
        if (pthread_detach(m_threads[i]) != 0)
        {
            delete[] m_threads;
            throw std::exception();
        }
    }
}

template <typename T>
threadpool<T>::~threadpool()
{
    delete[] m_threads;
    m_stop = true;
}

template <typename T>
bool threadpool<T>::append(T *request)
{
    m_queuelocker.lock();
    if (m_workqueue.size() > m_max_requests)
    {
        m_queuelocker.unlock();
        return false;
    }
    m_workqueue.push_back(request);
    m_queuelocker.unlock();
    m_queuestat.post();
    return true;
}

template <typename T>
void *threadpool<T>::worker(void *arg)
{
    threadpool *pool = (threadpool *)arg;
    pool->run();
    return pool;
}

template <typename T>
void threadpool<T>::run()
{
    while (!m_stop)
    {
        m_queuestat.wait();
        m_queuelocker.lock();
        if (m_workqueue.empty())
        {
            m_queuelocker.unlock();
            continue;
        }
        T *request = m_workqueue.front();
        m_workqueue.pop_front();
        m_queuelocker.unlock();
        if (!request)
        {
            continue;
        }

        connectionRAII mysqlcon(&request->mysql, m_connPool);
        request->process();
    }
}

#endif